package org.dromara.common.encrypt.core.encryptor;

import org.dromara.common.core.utils.StringUtils;
import org.dromara.common.encrypt.core.EncryptContext;
import org.dromara.common.encrypt.enumd.AlgorithmType;
import org.dromara.common.encrypt.enumd.EncodeType;
import org.dromara.common.encrypt.utils.EncryptUtils;


/**
 * @Author 刘武贵
 * @Date 2024/8/15 22:00
 * @Description 功能描述：sm2算法实现
 */
public class Sm2Encryptor extends AbstractEncryptor {

    /**
     * 加密上下文 用于encryptor传递必要的参数。
     */
    private final EncryptContext context;

    /**
     * 构造SM2加密器
     *
     * @param context 加密上下文，包含私钥和公钥
     * @throws IllegalArgumentException 如果私钥或公钥为空，则抛出此异常
     */
    public Sm2Encryptor(EncryptContext context) {
        super(context);
        // 从加密上下文中获取私钥和公钥
        String privateKey = context.getPrivateKey();
        String publicKey = context.getPublicKey();
        // 检查私钥和公钥是否为空，如果为空则抛出异常
        if (StringUtils.isAnyEmpty(privateKey, publicKey)) {
            throw new IllegalArgumentException("SM2公私钥均需要提供，公钥加密，私钥解密。");
        }
        // 将加密上下文保存为实例变量，以供后续使用
        this.context = context;
    }

    /**
     * 获得当前算法
     *
     * @return 当前算法类型为SM2
     */
    @Override
    public AlgorithmType algorithm() {
        return AlgorithmType.SM2;
    }

    /**
     * 加密
     *
     * 根据指定的编码类型加密给定的值
     * 此方法使用SM2加密算法，根据编码类型的不同，选择不同的加密方式
     *
     * @param value 待加密的字符串值
     * @param encodeType 加密时采用的编码类型，决定了使用哪种方式加密
     * @return 加密后的字符串
     */
    @Override
    public String encrypt(String value, EncodeType encodeType) {
        // 当编码类型为HEX时，使用SM2算法的Hex编码方式进行加密
        if (encodeType == EncodeType.HEX) {
            return EncryptUtils.encryptBySm2Hex(value, context.getPublicKey());
        } else {
            // 当编码类型不是HEX时，使用SM2算法的原始编码方式进行加密
            return EncryptUtils.encryptBySm2(value, context.getPublicKey());
        }
    }
    /**
     * 解密
     * 使用SM2加密算法对字符串进行解密
     *
     * 此方法主要负责接收一个经过SM2算法加密的字符串，并利用私钥对其进行解密
     * 它通过调用EncryptUtils工具类中的decryptBySm2方法实现解密操作
     *
     * @param value 待解密的字符串，该字符串应该是通过SM2算法加密的
     * @return 解密后的字符串
     */
    @Override
    public String decrypt(String value) {
        return EncryptUtils.decryptBySm2(value, context.getPrivateKey());
    }
}
